package com.qf.mybatisplus.utils;

import org.apache.commons.net.ftp.FTP;
import org.apache.commons.net.ftp.FTPClient;
import org.apache.commons.net.ftp.FTPFile;
import org.apache.commons.net.ftp.FTPReply;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.io.*;
import java.nio.charset.StandardCharsets;

/**
 * ftp上传下载工具类
 * <p>Title: FtpUtil</p>
 * <p>Description: </p>
 * <p>Company: www.itcast.com</p>
 * @author    入云龙
 * @date    2015年7月29日下午8:11:51
 * @version 1.0
 */
public class FtpUtils {
    private final static Logger logger = LoggerFactory.getLogger(FtpUtils.class);

    /**
     * Description: 向FTP服务器上传文件
     * @param host FTP服务器hostname
     * @param port FTP服务器端口
     * @param username FTP登录账号
     * @param password FTP登录密码
     * @param basePath FTP服务器基础目录
     * @param filePath FTP服务器文件存放路径。例如分日期存放：/2015/01/01。文件的路径为basePath+filePath
     * @param filename 上传到FTP服务器上的文件名
     * @param localFilePath 本地要上传的文件路径
     * @return 成功返回true，否则返回false
     */
    public static boolean uploadFile(String host, int port, String username, String password, String basePath,
                                     String filePath, String filename, String localFilePath) {
        boolean result = false;
        FTPClient ftp = new FTPClient();
        try {
            int reply;
            ftp.connect(host, port);// 连接FTP服务器
            // 如果采用默认端口，可以使用ftp.connect(host)的方式直接连接FTP服务器
            ftp.login(username, password);// 登录
            reply = ftp.getReplyCode();
            if (!FTPReply.isPositiveCompletion(reply)) {
                ftp.disconnect();
                logger.info("ftp11登陆失败");
                return result;
            }
            //切换到上传目录
            if (!ftp.changeWorkingDirectory(basePath+filePath)) {
                //如果目录不存在创建目录
                String[] dirs = filePath.split("/");
                String tempPath = basePath;
                for (String dir : dirs) {
                    if (null == dir || "".equals(dir)) continue;
                    tempPath += "/" + dir;
                    logger.info("777"+tempPath);
                    if (!ftp.changeWorkingDirectory(tempPath)) {
                        if (!ftp.makeDirectory(tempPath)) {
                            logger.info("ftp222创建文件夹失败");
                            return result;
                        } else {
                            ftp.changeWorkingDirectory(tempPath);
                        }
                    }
                }
            }
            //设置上传文件的类型为二进制类型
            ftp.setFileType(FTP.BINARY_FILE_TYPE);
            //上传文件
            File file = new File(localFilePath);
            InputStream input = new FileInputStream(file);
            if (!ftp.storeFile(filename, input)) {
                logger.info("ftp333上传文件失败");
                return result;
            }
            input.close();
            ftp.logout();
            result = true;
        } catch (IOException e) {
            e.printStackTrace();
        } finally {
            if (ftp.isConnected()) {
                try {
                    ftp.disconnect();
                } catch (IOException ioe) {
                }
            }
        }
        return result;
    }

    /**
     * Description: 从FTP服务器下载文件
     * @param host FTP服务器hostname
     * @param port FTP服务器端口
     * @param username FTP登录账号
     * @param password FTP登录密码
     * @param remotePath FTP服务器上的相对路径
     * @param fileName 要下载的文件名
     * @param localPath 下载后保存到本地的路径
     * @return
     */
    public static boolean downloadFile(String host, int port, String username, String password, String remotePath,
                                       String fileName, String localPath) {
        boolean result = false;
        FTPClient ftp = new FTPClient();
        try {
            int reply;
            ftp.connect(host, port);
            ftp.setAutodetectUTF8(true);
            // 如果采用默认端口，可以使用ftp.connect(host)的方式直接连接FTP服务器
            ftp.login(username, password);// 登录
            reply = ftp.getReplyCode();
            if (!FTPReply.isPositiveCompletion(reply)) {
                ftp.disconnect();
                return result;
            }
            String localCharset="GBK";
            if (FTPReply.isPositiveCompletion(ftp.sendCommand("OPTS UTF8", "ON"))) {
                localCharset = "UTF-8";
            }
            ftp.setControlEncoding(localCharset);
            logger.info("当前目录：{}", remotePath);
            Boolean res = ftp.changeWorkingDirectory( remotePath);// 转移到FTP服务器目录
            logger.info("切换目录，{}，{}","/"+ remotePath,res);
            logger.info("当前工作目录：{}",ftp.printWorkingDirectory());
            FTPFile[] fs = ftp.listFiles();
            for (FTPFile ff : fs) {
                logger.info("未处理前：{}",ff.getName());
                String fileNameOfFtp = new String(ff.getName().getBytes(StandardCharsets.ISO_8859_1), StandardCharsets.UTF_8);
                if (!fileNameOfFtp.equalsIgnoreCase(fileName)){
                    fileNameOfFtp = ff.getName();
                }
                logger.info("ftp文件名：{}，前端传的文件名：{}",fileNameOfFtp,fileName);
                if (fileNameOfFtp.equalsIgnoreCase(fileName)) {
                    File localFile = new File(localPath + "/" + fileName);
                    if (!localFile.getParentFile().exists()) {
                        localFile.getParentFile().mkdirs();
                    }
                    ftp.enterLocalPassiveMode();
                    OutputStream is = new FileOutputStream(localFile);
                    Boolean success = ftp.retrieveFile(new String(ff.getName().getBytes("UTF-8"),"iso-8859-1"), is);
                    logger.info("下载：{}",success);
                    is.close();
                    break;
                }
            }

            ftp.logout();
            result = true;
        } catch (IOException e) {
            e.printStackTrace();
        } finally {
            if (ftp.isConnected()) {
                try {
                    ftp.disconnect();
                } catch (IOException ioe) {
                }
            }
        }
        return result;
    }

}
